package com.tangsm.spring.boot.word.to.file.utli;

import com.itextpdf.text.Element;
import com.itextpdf.text.pdf.BaseFont;
import com.itextpdf.text.pdf.PdfContentByte;
import com.itextpdf.text.pdf.PdfGState;
import com.itextpdf.text.pdf.PdfReader;
import com.itextpdf.text.pdf.PdfStamper;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.FileOutputStream;
import java.io.IOException;

/**
 * 基于开源Libre Office文件处理工具类
 * @author tangsm
 */
public class LibreOfficeFileUtils {

    private static final Logger logger = LoggerFactory.getLogger(LibreOfficeFileUtils.class);

    /**
     * 利用libreOffice将office文档转换成pdf
     * @param filePath  源文件路径
     * @param outFolder    输出文件夹
     * @return 是否成功 true-成功, false-失败
     */
    public static boolean convertOffice2PDF(String filePath, String outFolder){
        logger.info("源文件路径：{}", filePath);
        logger.info("输出文件夹：{}", outFolder);

        long start = System.currentTimeMillis();

        // 成功标识
        boolean flag;
        // 系统命令
        String command;

        // 获取运行系统
        String osName = System.getProperty("os.name");
        if (osName.contains("Windows")) {
            // Win系统命令
            command = "soffice --headless --invisible --convert-to pdf:writer_pdf_Export " + filePath + " --outdir " + outFolder;
        }else {
            // Linux系统命令
            command = "libreoffice --headless --invisible --convert-to pdf:writer_pdf_Export " + filePath + " --outdir " + outFolder;
        }

        flag = executeLibreOfficeCommand(command);
        long end = System.currentTimeMillis();
        logger.debug("用时:{} ms", end - start);
        return flag;
    }

    /**
     * 执行command指令
     * @param command 系统命令
     * @return 是否成功 true-成功, false-失败
     */
    public static boolean executeLibreOfficeCommand(String command) {
        logger.info("开始进行转化.......");
        // Process可以控制该子进程的执行或获取该子进程的信息
        Process process;
        try {
            logger.debug("convertOffice2PDF cmd : {}", command);
            // exec()方法指示Java虚拟机创建一个子进程执行指定的可执行程序，并返回与该子进程对应的Process对象实例。
            process = Runtime.getRuntime().exec(command);
            // 下面两个可以获取输入输出流
            // InputStream errorStream = process.getErrorStream();
            // nputStream inputStream = process.getInputStream();
        } catch (IOException e) {
            logger.error(" convertOffice2PDF {} error", command, e);
            return false;
        }
        int exitStatus;
        try {
            // 等待子进程完成再往下执行，返回值是子线程执行完毕的返回值,返回0表示正常结束
            exitStatus = process.waitFor();
            // 接收执行完毕的返回值，第二种接受返回值的方法
            int i = process.exitValue();
            logger.debug("i----" + i);
        } catch (InterruptedException e) {
            logger.error("InterruptedException  convertOffice2PDF {}", command, e);
            return false;
        }

        if (exitStatus != 0) {
            logger.error("convertOffice2PDF cmd exitStatus {}", exitStatus);
        } else {
            logger.debug("convertOffice2PDF cmd exitStatus {}", exitStatus);
        }

        // 销毁子进程
        process.destroy();
        logger.info("转化结束.......");
        return true;
    }

    /**
     * 添加水印
     * @param pdfFilePath 原始文件路径
     * @param outputFilePath 输出文件路径
     * @param watermark 水印信息
     */
    public static void addPDFWater(String pdfFilePath, String outputFilePath, String watermark) {
        try {
            // 原PDF文件
            PdfReader reader = new PdfReader(pdfFilePath);
            // 输出的PDF文件内容
            PdfStamper stamper = new PdfStamper(reader, new FileOutputStream(outputFilePath));
            // 需要添加额外字体解决汉字乱码问题
            BaseFont baseFont = BaseFont.createFont("D:/tmp/simsun.ttf",BaseFont.IDENTITY_H,BaseFont.NOT_EMBEDDED);
            //BaseFont baseFont = BaseFont.createFont("C:/Windows/Fonts/simsun.ttf", BaseFont.IDENTITY_H, BaseFont.NOT_EMBEDDED);
//            BaseFont baseFont = BaseFont.createFont("/usr/share/fonts/chinese/simsun.ttf",BaseFont.IDENTITY_H,BaseFont.NOT_EMBEDDED);
            //BaseFont baseFont = BaseFont.createFont("STSong-Light", "UniGB-UCS2-H", BaseFont.EMBEDDED);
            PdfGState gs = new PdfGState();
            // 设置透明度
            gs.setFillOpacity(0.3f);
            gs.setStrokeOpacity(0.4f);
            //页数
            int totalPage = reader.getNumberOfPages() + 1;
            for (int i = 1; i < totalPage; i++) {
                // 内容上层
//			PdfContentByte content = stamper.getOverContent(i);
                // 内容下层
                PdfContentByte content = stamper.getUnderContent(i);
                content.beginText();
                // 字体添加透明度
                content.setGState(gs);
                // 添加字体大小等
                content.setFontAndSize(baseFont, 50);
                // 添加范围
                content.setTextMatrix(400, 780);
                // 具体位置 内容 旋转多少度 共360度
                content.showTextAligned(Element.ALIGN_CENTER, watermark, 100, 0, 300);
                content.showTextAligned(Element.ALIGN_CENTER, watermark, 200, 80, 300);
                content.showTextAligned(Element.ALIGN_CENTER, watermark, 300, 180, 300);
                content.showTextAligned(Element.ALIGN_CENTER, watermark, 400, 280, 300);
                content.endText();
            }
            // 关闭
            stamper.close();
            reader.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}
